how to get the average in python